"""
model name : 目标跟踪
file       : train.py
information:
    author : OuYang
    time   : 2025/7/19
"""

from ultralytics import YOLO


def train():
    model = YOLO("yolov8n.pt")

    model.train(
        data="data/otb_car24.yaml",
        epochs=100,
        imgsz=640,
        batch=16,
        name="yolov8n_otb_car24",
        lr0=0.01,
        lrf=0.001,
        warmup_epochs=3
    )

    print(f"训练完成")


if __name__ == '__main__':
    train()
